#-*-coding:utf-8-*-#

import os
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import geopandas as gpd
from shapely.geometry import Point, Polygon
import json

currentPath = os.getcwd()
os.chdir('C:/Users/owner/Desktop/Drought_NLP/Wildfire geo/data')

# Unzip Data_Fig1b_1 shape file of US states

usmap = gpd.read_file('tl_2017_us_state.shp')
usmap = usmap[~usmap.STUSPS.isin(['HI','AK'])]
usmap = usmap.to_crs("EPSG:2163")

# Unzip Data_Fig1b_2 shape file of US Drought Monitor

usdm = gpd.read_file('USDM_20120710.shp')
usdm = usdm.to_crs("EPSG:2163")

colors = ['#FFFF00','#FCD37F','#FFAA00','#E60000','#730000']


"""


month


"""

fig, ax = plt.subplots(figsize=(15, 15), dpi=400)
ax.axis('off')
usdm.plot(ax=ax, markersize=50, color=colors)
usmap.boundary.plot(ax=ax, color='black')
ax.set_xlim(-2400000, 2700000)
ax.set_ylim(-2400000, 890000)
plt.show()

